Pytorch Transformations To Augment Image Training Data